home *** CD-ROM | disk | FTP | other *** search
/ io Programmo 60 / IOPROG_60.ISO / soft / c++ / gsl-1.1.1-setup.exe / {app} / src / randist / geometric.c < prev    next >
Encoding:
C/C++ Source or Header  |  2000-05-05  |  1.6 KB  |  68 lines
  1. /* randist/geometric.c
  2.  * 
  3.  * Copyright (C) 1996, 1997, 1998, 1999, 2000 James Theiler, Brian Gough
  4.  * 
  5.  * This program is free software; you can redistribute it and/or modify
  6.  * it under the terms of the GNU General Public License as published by
  7.  * the Free Software Foundation; either version 2 of the License, or (at
  8.  * your option) any later version.
  9.  * 
  10.  * This program is distributed in the hope that it will be useful, but
  11.  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13.  * General Public License for more details.
  14.  * 
  15.  * You should have received a copy of the GNU General Public License
  16.  * along with this program; if not, write to the Free Software
  17.  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18.  */
  19.  
  20. #include <config.h>
  21. #include <math.h>
  22. #include <gsl/gsl_rng.h>
  23. #include <gsl/gsl_randist.h>
  24.  
  25. /* Geometric distribution (bernoulli trial with probability p) 
  26.  
  27.    prob(k) =  p (1 - p)^(k-1) for n = 1, 2, 3, ...
  28.  
  29.    It gives the distribution of "waiting times" for an event that
  30.    occurs with probability p. */
  31.  
  32. unsigned int
  33. gsl_ran_geometric (const gsl_rng * r, const double p)
  34. {
  35.   double u = gsl_rng_uniform_pos (r);
  36.  
  37.   unsigned int k;
  38.  
  39.   if (p == 1)
  40.     {
  41.       k = 1;
  42.     }
  43.   else
  44.     {
  45.       k = log (u) / log (1 - p) + 1;
  46.     }
  47.  
  48.   return k;
  49. }
  50.  
  51. double
  52. gsl_ran_geometric_pdf (const unsigned int k, const double p)
  53. {
  54.   if (k == 0)
  55.     {
  56.       return 0 ;
  57.     }
  58.   else if (k == 1)
  59.     {
  60.       return p ;
  61.     }
  62.   else
  63.     {
  64.       double P = p * pow (1 - p, k - 1.0);
  65.       return P;
  66.     }
  67. }
  68.